python oneline if else statment